package com.ruoyi.gd.khbjk.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.alibaba.excel.EasyExcel;
import com.fasterxml.jackson.databind.exc.InvalidFormatException;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.core.domain.PageQuery;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.gd.fj.domain.bo.ZFujianBo;
import com.ruoyi.gd.fj.domain.vo.ZFujianVo;
import com.ruoyi.gd.fj.service.IZFujianService;
import com.ruoyi.gd.khbjk.domain.vo.ImportZKhbjkVo;
import com.ruoyi.gd.khbjk.utils.KhbjkRowsListener;
import com.ruoyi.gd.wgd.domain.ZWgd;
import lombok.RequiredArgsConstructor;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import com.ruoyi.gd.khbjk.domain.bo.ZKhbjkBo;
import com.ruoyi.gd.khbjk.domain.vo.ZKhbjkVo;
import com.ruoyi.gd.khbjk.domain.ZKhbjk;
import com.ruoyi.gd.khbjk.mapper.ZKhbjkMapper;
import com.ruoyi.gd.khbjk.service.IZKhbjkService;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Collection;

/**
 * 客户报价库Service业务层处理
 *
 * @author ruoyi
 * @date 2024-07-01
 */
@RequiredArgsConstructor
@Service
public class ZKhbjkServiceImpl implements IZKhbjkService {

    private final ZKhbjkMapper baseMapper;

    @Autowired
    private IZFujianService fujianService;

    @Value("${upload.path}")
    private String uploadPath;

    /**
     * 查询客户报价库
     */
    @Override
    public ZKhbjkVo queryById(String id){
        return baseMapper.selectVoById(id);
    }

    /**
     * 查询客户报价库列表
     */
    @Override
    public TableDataInfo<ZKhbjkVo> queryPageList(ZKhbjkBo bo, PageQuery pageQuery) {
        LambdaQueryWrapper<ZKhbjk> lqw = buildQueryWrapper(bo);
        Page<ZKhbjkVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
        return TableDataInfo.build(result);
    }

    /**
     * 查询客户报价库列表
     */
    @Override
    public List<ZKhbjkVo> queryList(ZKhbjkBo bo) {
        LambdaQueryWrapper<ZKhbjk> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    private LambdaQueryWrapper<ZKhbjk> buildQueryWrapper(ZKhbjkBo bo) {
        StringUtils.trimEntityFields(bo);
        LambdaQueryWrapper<ZKhbjk> lqw = Wrappers.lambdaQuery();
        lqw.eq(StringUtils.isNotBlank(bo.getXh()), ZKhbjk::getXh, bo.getXh());
        lqw.like(StringUtils.isNotBlank(bo.getName()), ZKhbjk::getName, bo.getName());
        lqw.eq(StringUtils.isNotBlank(bo.getXhgg()), ZKhbjk::getXhgg, bo.getXhgg());
        lqw.eq(StringUtils.isNotBlank(bo.getZzcs()), ZKhbjk::getZzcs, bo.getZzcs());
        lqw.eq(StringUtils.isNotBlank(bo.getCcbh()), ZKhbjk::getCcbh, bo.getCcbh());
        lqw.eq(StringUtils.isNotBlank(bo.getGlbh()), ZKhbjk::getGlbh, bo.getGlbh());
        lqw.eq(bo.getNumber() != null, ZKhbjk::getNumber, bo.getNumber());
        lqw.eq(bo.getDj() != null, ZKhbjk::getDj, bo.getDj());
        lqw.eq(bo.getZj() != null, ZKhbjk::getZj, bo.getZj());
        lqw.eq(StringUtils.isNotBlank(bo.getFwfs()), ZKhbjk::getFwfs, bo.getFwfs());
        lqw.eq(StringUtils.isNotBlank(bo.getBz()), ZKhbjk::getBz, bo.getBz());
        lqw.eq(StringUtils.isNotBlank(bo.getKhid()), ZKhbjk::getKhid, bo.getKhid());
        lqw.like(StringUtils.isNotBlank(bo.getKhname()), ZKhbjk::getKhname, bo.getKhname());
        lqw.eq(StringUtils.isNotBlank(bo.getKhdz()), ZKhbjk::getKhdz, bo.getKhdz());
        lqw.eq(StringUtils.isNotBlank(bo.getKhlxr()), ZKhbjk::getKhlxr, bo.getKhlxr());
        lqw.eq(StringUtils.isNotBlank(bo.getKhdh()), ZKhbjk::getKhdh, bo.getKhdh());
        return lqw;
    }

    /**
     * 新增客户报价库
     */
    @Override
    public Boolean insertByBo(ZKhbjkBo bo) {
        ZKhbjk add = BeanUtil.toBean(bo, ZKhbjk.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setId(add.getId());
        }
        return flag;
    }

    /**
     * 修改客户报价库
     */
    @Override
    public Boolean updateByBo(ZKhbjkBo bo) {
        ZKhbjk update = BeanUtil.toBean(bo, ZKhbjk.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(ZKhbjk entity){
        //TODO 做一些数据校验,如唯一约束
    }

    /**
     * 批量删除客户报价库
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<String> ids, Boolean isValid) {
        if(isValid){
            //TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteBatchIds(ids) > 0;
    }

    // 导入数据
    @Override
    public int importData(ZKhbjkBo bo) {
        // 通过附件id找到附件
        if (StringUtils.isNotEmpty(bo.getFjid())) {
            List<ZFujianVo> zFujianVos = fujianService.queryListByIdOrdery(bo.getFjid());
            for (ZFujianVo zFujianVo : zFujianVos) {
                String fjpath = uploadPath + zFujianVo.getFjpath();
                KhbjkRowsListener khbjkRowsListener = new KhbjkRowsListener(20, bo);
                EasyExcel.read(fjpath, ImportZKhbjkVo.class, khbjkRowsListener).sheet().doRead();
                // 获取当前的读取的数据
                List<ImportZKhbjkVo> zkhbjkList = khbjkRowsListener.getZkhbjkList();
                // 判断当前的
                handleData(bo, zkhbjkList);
            }
        }
        return 1;
    }

    private void handleData(ZKhbjkBo bo, List<ImportZKhbjkVo> zkhbjkList) {
        String drlx = bo.getDrlx();
        for (ImportZKhbjkVo zKhbjkVo : zkhbjkList) {
            /**
             * 1.先通过的导入类型进行判断，当前是否通过哪些字段去查询数据
             * 2.如果查询到数据，如果有数据则将当前的数据进行更新，如果没有查询到数据，则添加数据
             */
            LambdaQueryWrapper<ZKhbjk> lqw = Wrappers.lambdaQuery(); // 构建当前的查询条件
            // 客户名称是必须要有的
            lqw.eq(StringUtils.isNotBlank(bo.getKhname()), ZKhbjk::getKhname, bo.getKhname());
            // 仪器名称
            if (drlx.contains("yqmc")) {
                lqw.eq(StringUtils.isNotBlank(zKhbjkVo.getName()), ZKhbjk::getName, zKhbjkVo.getName());
            }
            // 仪器型号
            if (drlx.contains("yqxh")) {
                lqw.eq(StringUtils.isNotBlank(zKhbjkVo.getXhgg()), ZKhbjk::getXhgg, zKhbjkVo.getXhgg());
            }
            // 仪器型号
            if (drlx.contains("ccbh")) {
                lqw.eq(StringUtils.isNotBlank(zKhbjkVo.getCcbh()), ZKhbjk::getCcbh, zKhbjkVo.getCcbh());
            }
            // 仪器型号
            if (drlx.contains("glbh")) {
                lqw.eq(StringUtils.isNotBlank(zKhbjkVo.getGlbh()), ZKhbjk::getGlbh, zKhbjkVo.getGlbh());
            }
            // 查询条件构建完毕，查询数据
            List<ZKhbjk> zKhbjks = baseMapper.selectList(lqw);
            // 判断当前的数据是否有数据，如果有数据，判断是一条数据还是多条数据，如果没有数据则进行添加
            if(zKhbjks.size() == 0) {
                ZKhbjk zKhbjk = BeanUtil.toBean(zKhbjkVo, ZKhbjk.class);
                baseMapper.insert(zKhbjk);
            }else if (zKhbjks.size() > 1) {
                for (ZKhbjk zKhbjk : zKhbjks) {
                    if (StringUtils.isNotEmpty(zKhbjkVo.getDj())) {
                        Double dj = Double.parseDouble(zKhbjkVo.getDj());
                        if (StringUtils.isNotEmpty(zKhbjkVo.getZj())) {
                            zKhbjk.setDj(dj);
                        }
                    }
                }
                // 将list数据批量修改
                baseMapper.updateBatchById(zKhbjks);
            }else {
                if (StringUtils.isNotEmpty(zKhbjkVo.getDj())) {
                    Double dj = Double.parseDouble(zKhbjkVo.getDj());
                    if (StringUtils.isNotEmpty(zKhbjkVo.getZj())) {
                        ZKhbjk zKhbjk = zKhbjks.get(0);
                        zKhbjk.setDj(dj);
                    }
                }
                // 将list数据批量修改
                baseMapper.updateBatchById(zKhbjks);
            }

        }
    }

}
